home *** CD-ROM | disk | FTP | other *** search
/ Personal Computer World 2009 February / PCWFEB09.iso / Software / Resources / Chat & Communication / Digsby build 37 / digsby_setup.exe / lib / getopt.pyo (.txt) < prev    next >
Python Compiled Bytecode  |  2008-10-13  |  4KB  |  162 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyo (Python 2.5)
  3.  
  4. __all__ = [
  5.     'GetoptError',
  6.     'error',
  7.     'getopt',
  8.     'gnu_getopt']
  9. import os
  10.  
  11. class GetoptError(Exception):
  12.     opt = ''
  13.     msg = ''
  14.     
  15.     def __init__(self, msg, opt = ''):
  16.         self.msg = msg
  17.         self.opt = opt
  18.         Exception.__init__(self, msg, opt)
  19.  
  20.     
  21.     def __str__(self):
  22.         return self.msg
  23.  
  24.  
  25. error = GetoptError
  26.  
  27. def getopt(args, shortopts, longopts = []):
  28.     opts = []
  29.     if type(longopts) == type(''):
  30.         longopts = [
  31.             longopts]
  32.     else:
  33.         longopts = list(longopts)
  34.     while args and args[0].startswith('-') and args[0] != '-':
  35.         if args[0] == '--':
  36.             args = args[1:]
  37.             break
  38.         
  39.         if args[0].startswith('--'):
  40.             (opts, args) = do_longs(opts, args[0][2:], longopts, args[1:])
  41.             continue
  42.         (opts, args) = do_shorts(opts, args[0][1:], shortopts, args[1:])
  43.     return (opts, args)
  44.  
  45.  
  46. def gnu_getopt(args, shortopts, longopts = []):
  47.     opts = []
  48.     prog_args = []
  49.     if isinstance(longopts, str):
  50.         longopts = [
  51.             longopts]
  52.     else:
  53.         longopts = list(longopts)
  54.     if shortopts.startswith('+'):
  55.         shortopts = shortopts[1:]
  56.         all_options_first = True
  57.     elif os.environ.get('POSIXLY_CORRECT'):
  58.         all_options_first = True
  59.     else:
  60.         all_options_first = False
  61.     while args:
  62.         if args[0] == '--':
  63.             prog_args += args[1:]
  64.             break
  65.         
  66.         if args[0][:2] == '--':
  67.             (opts, args) = do_longs(opts, args[0][2:], longopts, args[1:])
  68.             continue
  69.         if args[0][:1] == '-':
  70.             (opts, args) = do_shorts(opts, args[0][1:], shortopts, args[1:])
  71.             continue
  72.         if all_options_first:
  73.             prog_args += args
  74.             break
  75.             continue
  76.         prog_args.append(args[0])
  77.         args = args[1:]
  78.     return (opts, prog_args)
  79.  
  80.  
  81. def do_longs(opts, opt, longopts, args):
  82.     
  83.     try:
  84.         i = opt.index('=')
  85.     except ValueError:
  86.         optarg = None
  87.  
  88.     opt = opt[:i]
  89.     optarg = opt[i + 1:]
  90.     (has_arg, opt) = long_has_args(opt, longopts)
  91.     if has_arg:
  92.         if optarg is None:
  93.             if not args:
  94.                 raise GetoptError('option --%s requires argument' % opt, opt)
  95.             
  96.             optarg = args[0]
  97.             args = args[1:]
  98.         
  99.     elif optarg:
  100.         raise GetoptError('option --%s must not have an argument' % opt, opt)
  101.     
  102.     if not optarg:
  103.         pass
  104.     opts.append(('--' + opt, ''))
  105.     return (opts, args)
  106.  
  107.  
  108. def long_has_args(opt, longopts):
  109.     possibilities = _[1]
  110.     if opt in possibilities:
  111.         return (False, opt)
  112.     elif opt + '=' in possibilities:
  113.         return (True, opt)
  114.     
  115.     if len(possibilities) > 1:
  116.         raise GetoptError('option --%s not a unique prefix' % opt, opt)
  117.     
  118.     unique_match = possibilities[0]
  119.     has_arg = unique_match.endswith('=')
  120.     if has_arg:
  121.         unique_match = unique_match[:-1]
  122.     
  123.     return (has_arg, unique_match)
  124.  
  125.  
  126. def do_shorts(opts, optstring, shortopts, args):
  127.     while optstring != '':
  128.         opt = optstring[0]
  129.         optstring = optstring[1:]
  130.         if short_has_arg(opt, shortopts):
  131.             if optstring == '':
  132.                 if not args:
  133.                     raise GetoptError('option -%s requires argument' % opt, opt)
  134.                 
  135.                 optstring = args[0]
  136.                 args = args[1:]
  137.             
  138.             optarg = optstring
  139.             optstring = ''
  140.         else:
  141.             optarg = ''
  142.         opts.append(('-' + opt, optarg))
  143.     return (opts, args)
  144.  
  145.  
  146. def short_has_arg(opt, shortopts):
  147.     for i in range(len(shortopts)):
  148.         if shortopts[i] == shortopts[i]:
  149.             pass
  150.         elif shortopts[i] != ':':
  151.             return shortopts.startswith(':', i + 1)
  152.             continue
  153.     
  154.     raise GetoptError('option -%s not recognized' % opt, opt)
  155.  
  156. if __name__ == '__main__':
  157.     import sys
  158.     print getopt(sys.argv[1:], 'a:b', [
  159.         'alpha=',
  160.         'beta'])
  161.  
  162.